package storm;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;

/**
 * Redis 工具类
 * 
 * @author 李岩飞
 * @email eliyanfei@126.com
 * @date 2018年6月8日上午11:09:57
 */
public class JedisUtil {

	// Redis服务器IP
	public static String HOST = "192.168.56.10";

	// Redis的端口号
	public static int PORT = 6379;

	// 可用连接实例的最大数目，默认值为8；
	// 如果赋值为-1，则表示不限制；如果pool已经分配了maxActive个jedis实例，则此时pool的状态为exhausted(耗尽)。
	private static int MAX_ACTIVE = 1024;

	// 控制一个pool最多有多少个状态为idle(空闲的)的jedis实例，默认值也是8。
	private static int MAX_IDLE = 200;

	// 等待可用连接的最大时间，单位毫秒，默认值为-1，表示永不超时。如果超过等待时间，则直接抛出JedisConnectionException；
	private static int MAX_WAIT = 1000;

	// 超时时间
	private static int TIMEOUT = 60000;

	// 在borrow一个jedis实例时，是否提前进行validate操作；如果为true，则得到的jedis实例均是可用的；
	private static boolean TEST_ON_BORROW = true;

	private static JedisPool jedisPool = null;

	/**
	 * 初始化Redis连接池
	 */
	private static void initialPool() {
		try {
			JedisPoolConfig config = new JedisPoolConfig();
			config.setMaxTotal(MAX_ACTIVE);
			config.setMaxIdle(MAX_IDLE);
			config.setMaxWaitMillis(MAX_WAIT);
			config.setTestOnBorrow(TEST_ON_BORROW);
			config.setTestOnReturn(true);
			jedisPool = new JedisPool(config, HOST, PORT, TIMEOUT);
		} catch (Exception e) {
		}
	}

	/**
	 * 在多线程环境同步初始化
	 */
	private static void poolInit() {
		try {
			if (jedisPool == null) {
				initialPool();
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	public static synchronized Jedis getJedis() {
		Jedis jedis = null;
		for (int i = 0; i < 3; i++) {
			jedis = getJedis1();
			if (jedis != null)
				return jedis;
		}
		return jedis;
	}

	private static synchronized Jedis getJedis1() {
		if (jedisPool == null) {
			poolInit();
		}
		Jedis jedis = null;
		try {
			if (jedisPool != null) {
				jedis = jedisPool.getResource();
			}
		} catch (Exception e) {
			if (jedis != null)
				jedisPool.returnResource(jedis);
		}
		return jedis;
	}

	/**
	 * 释放jedis资源
	 * 
	 * @param jedis
	 */
	public static synchronized void returnResource(final Jedis jedis) {
		if (jedis != null && jedisPool != null) {
			try {
				jedisPool.returnResource(jedis);
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}
}